how to split an array into K subarrays